/*
 * Copyright (c) 2019 Nanjing Wind Information Co., Ltd. All Rights Reserved;
 * @Package com.wind.autotest.database
 * @version 1.0.0
 * @Date 2019年2月11日
 * @author jlliu.Farrior@wind.com.cn
 */
package wind.autotest.database.suport;

import lombok.extern.slf4j.Slf4j;

/**
 * @Description: <类功能描述>
 * @author jlliu.Farrior@wind.com.cn
 * @Date 2019年2月11日 上午11:27:46
 * 
 */
@Slf4j
public class DatabaseContextHolder {
	
	private static final ThreadLocal<DatabaseType> contextHolder = new ThreadLocal<>();

	public static void setDatabaseType(DatabaseType type) {
		contextHolder.set(type);
	}

	public static DatabaseType getDatabaseType() {
		return contextHolder.get();
	}
	
	public static void setDatabase(String datebaseSource){
		DatabaseType dataType = DatabaseType.stonetest;
		if ("测试站".equals(datebaseSource)) {
			dataType = DatabaseType.stonetest;
			
		} else if ("临时站".equals(datebaseSource)) {
			dataType = DatabaseType.stonetemp;
		} else {
			log.error("未知数据库站点：{}",datebaseSource);
		}
		
		setDatabaseType(dataType);
	}
}
